package com.warm.mini.wework.util;

import org.springframework.beans.BeansException;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.stereotype.Component;

/**
 * @Description Spring 获取 bean 工具类
 * @Author warmchuang
 * @Date 2022/2/10 19:46
 */
@Component
public class SpringContextUtil implements ApplicationContextAware {

    /**
     * 实现 ApplicationContextAware 接口后, 可以方便获取 ApplicationContext 中所有的 bean
     *
     * 通过 @Component 或者 XML 配置的方式注册 SpringContextUtil
     *
     * Spring 加载的时候会给实现了 ApplicationContextAware 接口的类自动注入 ApplicationContext 对象
     *   ** 源码: bean 加载过程
     *
     * 仅仅试用于：当前运行的代码和已启动的Spring代码处于同一个Spring上下文，否则获取到的ApplicationContext是空的
     */

    private static ApplicationContext applicationContext; // 使用静态成员变量

    @Override
    public void setApplicationContext(ApplicationContext context) throws BeansException {
        applicationContext = context;
    }

    /**
     * 取得存储在静态变量中的ApplicationContext.
     */
    public static ApplicationContext getApplicationContext() {
        checkApplicationContext();
        return applicationContext;
    }

    /**
     * 从静态变量ApplicationContext中取得Bean, 自动转型为所赋值对象的类型.
     */
    @SuppressWarnings("unchecked")
    public static <T> T getBean(String name) {
        return (T) getApplicationContext().getBean(name);
    }

    /**
     * 从静态变量ApplicationContext中取得Bean, 自动转型为所赋值对象的类型.
     */
    @SuppressWarnings("unchecked")
    public static <T> T getBean(Class<T> clazz) {
        return (T) getApplicationContext().getBeansOfType(clazz);
    }

    @SuppressWarnings("unchecked")
    public static <T> T getBean(String name, Class<T> clazz) {
        return (T) getApplicationContext().getBean(name, clazz);
    }

    /**
     * 清除applicationContext静态变量.
     */
    public static void cleanApplicationContext() {
        applicationContext = null;
    }

    private static void checkApplicationContext() {
        if (applicationContext == null) {
            throw new IllegalStateException("applicationContext未注入");
        }
    }

}
